import pdb

import numpy as np
from torch import Tensor

from mmengine.structures import BaseDataElement



class SequenceData(BaseDataElement):
    def set_seq(self, value: Tensor|np.ndarray):
        assert value.ndim >= 2, f"Value must have at least 2 dimensions (Channel, Step), but got {value.ndim}"
        super().__setattr__('data', value)


